CUDA_VISIBLE_DEVICES=7 python supervised_ds_supr.py \
--n_train 3 \
--data_dir ./data/CANDI-128-160-norm \
--out_dir ./save_seg/supervised \
--epochs 40 \
--batch_size 16 \
--resolution 160 \
--n_class 16 \
--shot 103 \
--aug

# bs = 32: 10000 + 11200*3 mem use; snap = 25 ticks = 100kimg = 34min
# CUDA_VISIBLE_DEVICES=0,1,4,5 python train.py \
# --outdir save \
# --data data/US/images \
# --kimg 25000 \
# --batch 32 \
# --snap 25 \
# --mirror True \
# --gpus=4 \
# --seed 0

# CUDA_VISIBLE_DEVICES=0,1,4,5 python train.py \
# --outdir save \
# --data data/GAN_CANDI \
# --kimg 20000 \
# --batch 32 \
# --snap 25 \
# --mirror True \
# --gpus=4 \
# --DiffAugment '' \
# --aug noaug \
# --seed 0

# bs = 32: 10000 + 11200*3 mem use; snap = 25 ticks = 100kimg = 34min
# CUDA_VISIBLE_DEVICES=1,2,3,6 python train.py \
# --outdir save \
# --data data/CelebA-HQ-img-256 \
# --kimg 25000 \
# --batch 32 \
# --snap 25 \
# --mirror True \
# --gpus=4 \
# --seed 0


# CUDA_VISIBLE_DEVICES=3,6 python train.py \
# --outdir save \
# --data data/CelebA-HQ-img \
# --kimg 25000 \
# --batch 4 \
# --snap 25 \
# --mirror True \
# --gpus=2 \
# --seed 0
